update to get tests passing