fixed invalid return type