Fixing #150

Wrong trait names was used