Add defense code for wrong use cases