fixes return type
simplified syntax a bit