bug fixes :shit: