Small QoL bug fix