Fixed a typehint
Fixed a bug that caused a test to fail