Readd missing part of test