fixes after testing

--amend