some bug fixes -.-