Fix type, should sort out #1006