defo a nicer way to handle this situation