Fix for initial trace being extracted from incorrect axis