Refactored by adding mock of the plane at the top