fix failing test cases due to wrong typing