fixing after tests