Passing tests with n_states <= 4!!!!!