Fix tests section