Fix/ return type error