fixed a few missing things