Fix compatibility problem