Fixed wrong usage of trait