Bug fixes, test update