Move smoke tests to function