fix returned type